package master;


import java.util.List;

/**
 *
 * K个一组翻转链表
 *
 * 在整个过程中一共需要保存四个节点的指针，
 *      需要被反转的子链表的头尾节点head，tail
 *      需要被反转的子链表的头结点的前一个节点pre
 *      需要被反转的子链表的尾节点的下一个节点nex
 *
 * @author chenyw
 * @date 2022/7/24 11:42
 */
public class M25reverseKGroup {
    public ListNode reverseKGroup(ListNode head, int k) {
        ListNode hair = new ListNode(0);
        hair.next = head;
        //记录需要反转的链表得头结点的上一个节点
        ListNode pre = hair;
        while (head != null) {
            ListNode tail = pre;
            //判断后面是否够一组k个
            for (int i = 0; i < k; i++) {
                tail = tail.next;
                if (tail == null) {
                    return hair.next;
                }
            }
            //记录需要反转的链表的尾节点的下一个节点
            ListNode nex = tail.next;
            //反转链表, 因为要记录反转后子链表的头尾节点。所以递归不合适
            ListNode[] lists = reverse(head, tail);
            //将反转后的链表连接上
            head = lists[0];
            tail = lists[1];
            pre.next = head;
            tail.next = nex;
            head = nex;
            pre = tail;
        }
        return hair.next;

    }

    private ListNode[] reverse(ListNode head, ListNode tail) {
        // 相当于给pre付个初始值，这养写可以利用已有的对象省空间
        ListNode pre = tail.next;
        ListNode p = head;
        while (pre != tail) {
            ListNode nex = p.next;
            p.next = pre;
            pre = p;
            p = nex;
        }
        return new ListNode[] {tail, head};
    }
}
